home *** CD-ROM | disk | FTP | other *** search
/ Windows 95 API Bible / Windows 95 API Bible 3 Disc Set.iso / Win32 API Bible Book 2 of 3.iso / chapter4 / rebootsy.c < prev    next >
C/C++ Source or Header  |  1996-03-06  |  6KB  |  229 lines
  1.  
  2. #include <windows.h>  
  3. #include "rebootsy.h"  
  4. #include "commctrl.h"
  5.  
  6.  
  7. #if defined (WIN32)
  8.     #define IS_WIN32 TRUE
  9. #else
  10.     #define IS_WIN32 FALSE
  11. #endif
  12.  
  13. #define IS_NT      IS_WIN32 && (BOOL)(GetVersion() < 0x80000000)
  14. #define IS_WIN32S  IS_WIN32 && (BOOL)(!(IS_NT) && (LOBYTE(LOWORD(GetVersion()))<4))
  15. #define IS_WIN95   (BOOL)(!(IS_NT) && !(IS_WIN32S)) && IS_WIN32
  16.  
  17. HINSTANCE hInst;   // current instance
  18.  
  19. LPCTSTR lpszAppName = "MyApp";
  20. LPCTSTR lpszTitle   = "PropSheet_RebootSystem"; 
  21.  
  22.  
  23. BOOL RegisterWin95( CONST WNDCLASS* lpwc );
  24.  
  25.  
  26. int APIENTRY WinMain( HINSTANCE hInstance, HINSTANCE hPrevInstance,
  27.                       LPTSTR lpCmdLine, int nCmdShow)
  28. {
  29.    MSG      msg;
  30.    HWND     hWnd; 
  31.    WNDCLASS wc;
  32.  
  33.    wc.style         = CS_HREDRAW | CS_VREDRAW;
  34.    wc.lpfnWndProc   = (WNDPROC)WndProc;       
  35.    wc.cbClsExtra    = 0;                      
  36.    wc.cbWndExtra    = 0;                      
  37.    wc.hInstance     = hInstance;              
  38.    wc.hIcon         = LoadIcon (hInstance, lpszAppName); 
  39.    wc.hCursor       = LoadCursor(NULL, IDC_ARROW);
  40.    wc.hbrBackground = (HBRUSH)(COLOR_WINDOW+1);
  41.    wc.lpszMenuName  = lpszAppName;              
  42.    wc.lpszClassName = lpszAppName;              
  43.  
  44.    if ( IS_WIN95 )
  45.    {
  46.       if ( !RegisterWin95( &wc ) )
  47.          return( FALSE );
  48.    }
  49.    else if ( !RegisterClass( &wc ) )
  50.       return( FALSE );
  51.  
  52.    hInst = hInstance; 
  53.  
  54.    hWnd = CreateWindow( lpszAppName, 
  55.                         lpszTitle,    
  56.                         WS_OVERLAPPEDWINDOW, 
  57.                         CW_USEDEFAULT, 0, 
  58.                         CW_USEDEFAULT, 0,  
  59.                         NULL,              
  60.                         NULL,              
  61.                         hInstance,         
  62.                         NULL               
  63.                       );
  64.  
  65.    if ( !hWnd ) 
  66.       return( FALSE );
  67.  
  68.    ShowWindow( hWnd, nCmdShow ); 
  69.    UpdateWindow( hWnd );         
  70.  
  71.    while( GetMessage( &msg, NULL, 0, 0) )   
  72.    {
  73.       TranslateMessage( &msg ); 
  74.       DispatchMessage( &msg );  
  75.    }
  76.  
  77.    return( msg.wParam ); 
  78. }
  79.  
  80.  
  81. BOOL RegisterWin95( CONST WNDCLASS* lpwc )
  82. {
  83.    WNDCLASSEX wcex;
  84.  
  85.    wcex.style         = lpwc->style;
  86.    wcex.lpfnWndProc   = lpwc->lpfnWndProc;
  87.    wcex.cbClsExtra    = lpwc->cbClsExtra;
  88.    wcex.cbWndExtra    = lpwc->cbWndExtra;
  89.    wcex.hInstance     = lpwc->hInstance;
  90.    wcex.hIcon         = lpwc->hIcon;
  91.    wcex.hCursor       = lpwc->hCursor;
  92.    wcex.hbrBackground = lpwc->hbrBackground;
  93.    wcex.lpszMenuName  = lpwc->lpszMenuName;
  94.    wcex.lpszClassName = lpwc->lpszClassName;
  95.  
  96.    // Added elements for Windows 95.
  97.    //...............................
  98.    wcex.cbSize = sizeof(WNDCLASSEX);
  99.    wcex.hIconSm = LoadImage(wcex.hInstance, lpwc->lpszClassName, 
  100.                             IMAGE_ICON, 16, 16,
  101.                             LR_DEFAULTCOLOR );
  102.             
  103.    return RegisterClassEx( &wcex );
  104. }
  105.  
  106.  
  107. LRESULT CALLBACK PageProc( HWND hDlg, UINT message,        
  108.                            WPARAM wParam, LPARAM lParam )
  109. {
  110.    switch (message) 
  111.    {
  112.        case WM_INITDIALOG: 
  113.              return (TRUE);
  114.  
  115.        case WM_NOTIFY :
  116.              if ( ((NMHDR*)lParam)->code == PSN_APPLY )
  117.              {
  118.                 // Apply changes.
  119.                 if ( IsDlgButtonChecked( hDlg, IDC_REBOOT ) )
  120.                    PropSheet_RebootSystem( GetParent( hDlg ) );
  121.  
  122.                 if ( IsDlgButtonChecked( hDlg, IDC_RESTART ) )
  123.                    PropSheet_RestartWindows( GetParent( hDlg ) );
  124.              }
  125.              break;
  126.    }
  127.  
  128.    return( FALSE );
  129. }
  130.  
  131.  
  132. VOID DisplayProperties( HWND hWnd )
  133. {
  134.    PROPSHEETPAGE   psp;
  135.    PROPSHEETHEADER psh;
  136.    HPROPSHEETPAGE  hpsp;
  137.  
  138.    psp.dwSize      = sizeof( PROPSHEETPAGE );
  139.    psp.dwFlags     = PSP_DEFAULT;
  140.    psp.hInstance   = hInst;
  141.    psp.pszTemplate = "RADIOPAGE";
  142.    psp.pfnDlgProc  = (DLGPROC)PageProc;
  143.    hpsp = CreatePropertySheetPage( &psp );
  144.  
  145.    memset( &psh, 0, sizeof( PROPSHEETHEADER ) );
  146.    psh.dwSize     = sizeof( PROPSHEETHEADER );
  147.    psh.dwFlags    = PSH_USEICONID | PSH_NOAPPLYNOW;
  148.    psh.hInstance  = hInst;
  149.    psh.hwndParent = hWnd;
  150.    psh.pszIcon    = "SMALL";
  151.    psh.nPages     = 1;
  152.    psh.phpage     = &hpsp;
  153.    psh.pszCaption = "Properties";
  154.  
  155.    switch( PropertySheet( &psh ) )
  156.    {
  157.       case ID_PSREBOOTSYSTEM :
  158.              ExitWindowsEx( EWX_FORCE | EWX_REBOOT, 0 );
  159.              DestroyWindow( hWnd );
  160.              break;
  161.  
  162.       case ID_PSRESTARTWINDOWS :
  163.              ExitWindowsEx( EWX_FORCE | EW_RESTARTWINDOWS , 0 );
  164.              DestroyWindow( hWnd );
  165.              break;
  166.    }
  167. }
  168.  
  169.  
  170. LRESULT CALLBACK WndProc( HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam )
  171. {
  172.    switch( uMsg )
  173.    {
  174.       case WM_CREATE :
  175.               InitCommonControls();
  176.               break;
  177.  
  178.       case WM_COMMAND :
  179.               switch( LOWORD( wParam ) )
  180.               {
  181.                  case IDM_TEST :
  182.                         DisplayProperties( hWnd );
  183.                         break;
  184.  
  185.                  case IDM_ABOUT :
  186.                         DialogBox( hInst, "AboutBox", hWnd, (DLGPROC)About );
  187.                         break;
  188.  
  189.                  case IDM_EXIT :
  190.                         DestroyWindow( hWnd );
  191.                         break;
  192.               }
  193.               break;
  194.       
  195.       case WM_DESTROY :
  196.               PostQuitMessage(0);
  197.               break;
  198.  
  199.       default :
  200.             return( DefWindowProc( hWnd, uMsg, wParam, lParam ) );
  201.    }
  202.  
  203.    return( 0L );
  204. }
  205.  
  206.  
  207. LRESULT CALLBACK About( HWND hDlg,           
  208.                         UINT message,        
  209.                         WPARAM wParam,       
  210.                         LPARAM lParam)
  211. {
  212.    switch (message) 
  213.    {
  214.        case WM_INITDIALOG: 
  215.                return (TRUE);
  216.  
  217.        case WM_COMMAND:                              
  218.                if (   LOWORD(wParam) == IDOK         
  219.                    || LOWORD(wParam) == IDCANCEL)    
  220.                {
  221.                        EndDialog(hDlg, TRUE);        
  222.                        return (TRUE);
  223.                }
  224.                break;
  225.    }
  226.  
  227.    return (FALSE); 
  228. }
  229.